package com.fmm.medium;

import java.util.ArrayList;
import java.util.List;

public class RecoverIpAddress {

    /**
     * @param s 给定一个数字组成的字符串
     * @return 返回所有可能的IP地址
     */
    public static List<String> restoreIpAddresses(String s) {
        return restoreIpAddresses(s, 0, 4);
    }

    /**
     * @param s 原始字符串
     * @param index 从 index 开始
     * @param k 从 index 开始之后的子字符串需要分割成 k 份
     * @return 所有可能的列表
     */
    private static List<String> restoreIpAddresses(String s, int index, int k) {
        List<String> result = new ArrayList<>();
        // 边界条件
        if (k == 1) {
            if (index >= s.length()) {
                return result;
            }
            // 最后一部分字符串是否合法
            String lastPart = s.substring(index);
            if (isValid(lastPart)) {
                result.add(lastPart);
            }
            return result;
        }
        // 尝试分割从 index 开始，一个字符，两个字符，三个字符组成当前循环的字符串，与后续递归结果组合
        for (int i = index; i < index + 3 && i < s.length(); i++) {
            String substring;
            if (i + 1 >= s.length()) {
                substring = s.substring(index);
            } else {
                substring = s.substring(index, i + 1);
            }
            // 当前字符串如何合法，尝试与后续结果组合
            if (isValid(substring)) {
                // 后续结果
                List<String> subResult = restoreIpAddresses(s, i + 1, k - 1);
                if (subResult.isEmpty()) {
                    continue;
                }
                // 当前字符串与后续结果组合
                for (String sub : subResult) {
                    result.add(substring + "." + sub);
                }
            }
        }
        return result;
    }

    /**
     * 一段 ip 地址是否符合要求
     * */
    private static boolean isValid(String s) {
        // 长度为 0 不符合
        if (s.length() == 0) {
            return false;
        }
        // 前导 0 不符合
        if (s.length() > 1 && s.charAt(0) == '0') {
            return false;
        }
        // 长度大于 3 不符合
        if (s.length() > 3) {
            return false;
        }
        // 数字在 0 - 255 范围内
        Integer num = Integer.parseInt(s);
        return num <= 255 && num >= 0;
    }

    public static void main(String[] args) {
        List<String> result = restoreIpAddresses("0279245587303");
        for (String s : result) {
            System.out.println(s);
        }
    }
}
